import random

def generate_data(n=10, k=5, max_reward=100):
    reward1 = [random.randint(1, max_reward) for _ in range(n)]
    reward2 = [random.randint(1, max_reward) for _ in range(n)]
    return reward1, reward2, k

if __name__ == "__main__":
    n = 10
    k = 5
    reward1, reward2, k = generate_data(n, k)
    print("reward1 =", reward1)
    print("reward2 =", reward2)
    print("k =", k)